ResKV: Reconstructing Omitted Attention Contributions for Fixed-Budget KV Cache Compression
ResKV:固定预算下通过残差缓存重构被忽略注意力贡献的KV缓存压缩
🧭 固定预算KV缓存压缩
University of Electronic Science and Technology of China
将KV缓存预算分为精确主缓存与紧凑残差缓存,同框归一化复原被丢弃token的注意力分子分母贡献,广泛提升长上下文性能。
📌 工作参考价值(3 点)
- 在电商推荐的长用户行为序列建模中,若使用LLM生成式推荐,KV缓存膨胀是瓶颈,可借鉴ResKV的残差缓存思路:保留核心token作为精确主缓存,用紧凑残差统计量重构被丢弃token的注意力贡献,从而在内存受限下保持长上下文的注意力完整性。
- 工程实现上,残差缓存的构建可通过离线验证决定每层每头的残差预算分配,解码时动态门控按query调整残差贡献,这种在线-离线协同策略可以迁移到推荐系统中的搜索/浏览长序列压缩,平衡效果与延迟。
- 残差缓存参与同一softmax归一化,而不是事后修正,这一设计确保注意力分数的分子和分母质量都得到恢复,相比简单的eviction或合并方法,对下游生成质量更友好,适合高精度要求的选品文案生成、个性化搜索词推荐等场景。